TRAINING_PARAMS = \
{
    "model_params": {
        "backbone_name": "darknet_53",
        "backbone_pretrained": "",  # set empty to disable
    },
    "yolo": {
        "anchors": [[[152, 98], [201, 149], [210, 50]],
                    [[67, 222], [78, 150], [134, 214]],
                    [[40, 90], [52, 121], [62, 126]]],
        "classes": 80,
    },
    "batch_size": 16,
    # iou_thres = 0.5, conf_thres = 0.25, mAP = 0.939
    # iou_thres = 0.5, conf_thres = 0.45, mAP = 0.746
    # iou_thres = 0.25, conf_thres = 0.25, mAP = 0.973
    "iou_thres": 0.25,
    "val_path": "/mnt/sda1/hgl/indicator/indicator_image_416_sorted.txt",
    "val_label_path": "/mnt/sda1/hgl/indicator/label_normalize/indicator_indicator_sorted.txt",
    "epochs": 400,
    "img_h": 416,
    "img_w": 416,
    "parallels": [1],                             # config GPU device
    "pretrain_snapshot": "../bobliu_yolov3/darknet_53/size416x416_try5/20190328181731/flat_05_model.pth",    # load checkpoint
}
